hackerrank problem solving in python